package com.zhz.leetcode;

import com.zhz.leetcode.entity.ListNode;
import com.zhz.leetcode.entity.TreeNode;
import com.zhz.leetcode.utils.LinkedListUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * 2. 两数相加
 * 给你两个非空 的链表，表示两个非负的整数。它们每位数字都是按照逆序的方式存储的，并且每个节点只能存储一位数字。
 * 请你将两个数相加，并以相同形式返回一个表示和的链表。
 * 你可以假设除了数字 0 之外，这两个数都不会以 0开头。
 * <p>
 * 示例 1：
 * 输入：l1 = [2,4,3], l2 = [5,6,4]
 * 输出：[7,0,8]
 * 解释：342 + 465 = 807.
 * <p>
 * 示例 2：
 * 输入：l1 = [0], l2 = [0]
 * 输出：[0]
 * <p>
 * 示例 3：
 * 输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
 * 输出：[8,9,9,9,0,0,0,1]
 * <p>
 * 提示：
 * 每个链表中的节点数在范围 [1, 100] 内
 * 0 <= Node.val <= 9
 * 题目数据保证列表表示的数字不含前导零
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/add-two-numbers
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class AddTwoNumbers2 {

    public static void main(String[] args) {

        AddTwoNumbers2 m = new AddTwoNumbers2();
        ListNode node1 = LinkedListUtils.createListNode(new int[]{2,4,3});
        ListNode node2 = LinkedListUtils.createListNode(new int[]{5,6,7});
        ListNode res = m.addTwoNumbers(node1, node2);
        LinkedListUtils.printList(res);
    }

    /**
     *
     */
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode resNode = new ListNode(0);
        ListNode currNode = resNode;
        boolean needPlusOne = false;
        while (l1 != null || l2 != null) {
            int num1 = l1 == null ? 0 : l1.val;
            int num2 = l2 == null ? 0 : l2.val;
            int tmp = num1 + num2 + (needPlusOne ? 1 : 0);
            //这边只取个位数
            needPlusOne = tmp > 9;
            currNode.next = new ListNode(tmp % 10);
            currNode = currNode.next;
            if (l2 != null) {
                l2 = l2.next;
            }
            if (l1 != null) {
                l1 = l1.next;
            }
        }
        if(needPlusOne){
            currNode.next = new ListNode(1);
        }
        return resNode.next;
    }

}
